package com.xu.cloud.shorturl.converter;

import com.xu.cloud.shorturl.pojo.domain.ShortUrlInfoDO;
import com.xu.cloud.shorturl.pojo.dto.UrlInfoDTO;

import java.util.List;
import java.util.stream.Collectors;

/**
 * @author xuguan
 * @since 2024/3/10
 */
public class UrlInfoDTOConverter {
    public static UrlInfoDTO convertShortUrlInfoDO(ShortUrlInfoDO shortUrlInfoDO) {
        final UrlInfoDTO dto = new UrlInfoDTO();
        dto.setShortUrl(shortUrlInfoDO.getShortUrl());
        dto.setOriginUrl(shortUrlInfoDO.getOriginUrl());
        return dto;
    }

    public static List<UrlInfoDTO> convertShortUrlInfoDOList(List<ShortUrlInfoDO> shortUrlInfoDO) {
        return shortUrlInfoDO.stream()
                .map(UrlInfoDTOConverter::convertShortUrlInfoDO)
                .collect(Collectors.toList());
    }
}
